# @summary A short summary of the purpose of this class
#
# A description of what this class does
#
# @example
#   include nvgpu::yum
class nvgpu::yum (
  $nvdriver_version="450.191.01-1.el7.x86_64",
){
  class{'yum::plugin::versionlock':
    clean => true,
  }
  yum::versionlock { "3:nvidia-driver-latest-$nvdriver_version":
    ensure => present,
  }
  yum::versionlock { "0:cuda-drivers-$nvdriver_version":
    ensure => present,
  }
  yum::versionlock { "3:3:kmod-nvidia-latest-dkms-$nvdriver_version":
    ensure => present,
  }
}
